package com.db.onlinedb.utils;

import java.sql.Connection;
import java.util.HashMap;
import java.util.Map;

/**
 * 连接映射，host和连接名 <<>>> connection
 */
public class ConnectionMap {
    private static Map<String, Conn> map = new HashMap<>();

    /**
     * 私有构造方法
     */
    private ConnectionMap() {
    }

    /**
     * 全局获取Conn方法
     *
     * @param lh host和连接名
     * @return Conn
     */
    public static Connection findConn(String lh) {
        return map.get(lh).requestConnection();
    }

    /**
     * 添加映射
     *
     * @param linkName 会话
     * @param conn     连接
     */
    public static void addMap(String lh, Conn conn) {
        map.put(lh, conn);
    }


    public static void removeMap(String lh) {
        map.remove(lh);
    }
}
